package com.tanhua.dubbo.server.service;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

/**
 * 原理:使用redis的自增长的类型,实现自增长的类型
 */
@Service
public class IDService {

    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    /**
     * 生成自增长的id
     * @return
     */
    public Long createId(String type,String objectId) {

        //将type转成大写
        type= StringUtils.upperCase(type);
        String hashKey = "TANHUA_HASH_ID_" +type;
        //入锅ObjectId已经存在 就返回对应的id

        if (this.redisTemplate.opsForHash().hasKey(hashKey,objectId)) {
            //因为返回值是Long类型 所以需要转化
            return Long.valueOf(this.redisTemplate.opsForHash().get(hashKey,objectId).toString());
        }
        String key = "TANHUA_ID_" + type;
        //实现自增长  increment
        Long id= this.redisTemplate.opsForValue().increment(key);

        //将生成的id写入到hash表中
        this.redisTemplate.opsForHash().put(hashKey,objectId,id);
        return id;
    }
}
